# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/11 15:13 
@Author  : ZhangShenao 
@File    : re_generate.py 
@Desc    : generator只能迭代一次

有限元素的generator只能迭代一次,重复迭代会抛出StopIteration异常
"""


def square_generator(n):
    """平方生成器,生成1~n的平方数"""
    for i in range(n + 1):
        yield i ** 2


# 有限元素的generator只能迭代一次,重复迭代会抛出StopIteration异常
squares = square_generator(10)
for i in range(10):
    print(next(squares))
